package mark_2.io;

import java.util.HashMap;
import java.util.Map;

public class ImageManager {
    private Map<String, String> imageMap = new HashMap<>(); // 键：图片名，值：本地路径

    // 下载图片并加入管理
    public boolean downloadImage(String url) {
        // 从URL中提取图片名
        String[] parts = url.split("/");
        String imageName = parts[parts.length - 1];
        // 确保“images”目录存在
        FileUtils.ensureDirectoryExists("images");
        String savePath = "images/" + imageName;

        if (FileUtils.saveImage(url, savePath)) {
            imageMap.put(imageName, savePath); // 记录图片名与路径的映射
            FileUtils.appendLog(url, savePath); // 写日志
            return true;
        }
        return false;
    }

    // 列出所有已下载图片
    public void listImages() {
        if (imageMap.isEmpty()) {
            System.out.println("暂无已下载的图片～");
            return;
        }
        System.out.println("已下载图片列表：");
        for (Map.Entry<String, String> entry : imageMap.entrySet()) {
            System.out.println("图片名：" + entry.getKey() + " | 本地路径：" + entry.getValue());
        }
    }

    // 按名称搜索图片路径
    public String searchImage(String name) {
        return imageMap.get(name);
    }
}

